import numpy as np
from sklearn.datasets import make_regression
from scipy.spatial.distance import norm
from itertools import product
from collections import OrderedDict
from plotly.graph_objs import *
import plotly.tools as tls
from plotly.offline import download_plotlyjs, init_notebook_mode, plot, iplot
init_notebook_mode()
import time

from plot_helpers import *

X, y = make_regression(n_samples = int(1e5), n_features = 2, n_informative=2, random_state=0, noise=10)
X = (X - X.mean(axis=0))/X.std()

data = Scatter3d(
        x=X[:,0],
        y=X[:,1],
        z=y,
        mode='markers',
        marker={'size':1}
)

layout = Layout(
    scene = dict(
        xaxis = dict(title='x0'),
        yaxis = dict(title='x1'),
        zaxis = dict(title='y')
        )
)

figure_ = Figure(data=[data], layout=layout)

iplot(figure_)